// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_CHILD_SERVICE_WORKER_WEB_SERVICE_WORKER_IMPL_H_
#define CONTENT_CHILD_SERVICE_WORKER_WEB_SERVICE_WORKER_IMPL_H_

#include <memory>
#include <vector>

#include "base/compiler_specific.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/strings/string16.h"
#include "content/common/content_export.h"
#include "third_party/WebKit/public/platform/WebMessagePortChannel.h"
#include "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorker.h"
#include "third_party/WebKit/public/web/WebFrame.h"

namespace blink {
class WebServiceWorkerProxy;
}

namespace content {

class ServiceWorkerHandleReference;
class ThreadSafeSender;

// Each instance corresponds to one ServiceWorker object in JS context, and
// is held by ServiceWorker object in Blink's C++ layer via
// WebServiceWorker::Handle.
//
// Each instance holds one ServiceWorkerHandleReference so that
// corresponding ServiceWorkerHandle doesn't go away in the browser process
// while the ServiceWorker object is alive.
class CONTENT_EXPORT WebServiceWorkerImpl
    : NON_EXPORTED_BASE(public blink::WebServiceWorker),
      public base::RefCounted<WebServiceWorkerImpl> {
public:
    WebServiceWorkerImpl(std::unique_ptr<ServiceWorkerHandleReference> handle_ref,
        ThreadSafeSender* thread_safe_sender);

    void OnStateChanged(blink::WebServiceWorkerState new_state);

    // blink::WebServiceWorker overrides.
    void setProxy(blink::WebServiceWorkerProxy* proxy) override;
    blink::WebServiceWorkerProxy* proxy() override;
    blink::WebURL url() const override;
    blink::WebServiceWorkerState state() const override;
    void postMessage(blink::WebServiceWorkerProvider* provider,
        const blink::WebString& message,
        const blink::WebSecurityOrigin& source_origin,
        blink::WebMessagePortChannelArray* channels) override;
    void terminate() override;

    // Creates WebServiceWorker::Handle object that owns a reference to the given
    // WebServiceWorkerImpl object.
    static std::unique_ptr<blink::WebServiceWorker::Handle> CreateHandle(
        const scoped_refptr<WebServiceWorkerImpl>& worker);

private:
    friend class base::RefCounted<WebServiceWorkerImpl>;
    ~WebServiceWorkerImpl() override;

    std::unique_ptr<ServiceWorkerHandleReference> handle_ref_;
    blink::WebServiceWorkerState state_;
    scoped_refptr<ThreadSafeSender> thread_safe_sender_;
    blink::WebServiceWorkerProxy* proxy_;

    DISALLOW_COPY_AND_ASSIGN(WebServiceWorkerImpl);
};

} // namespace content

#endif // CONTENT_CHILD_SERVICE_WORKER_WEB_SERVICE_WORKER_IMPL_H_
